#region Copyright

// The contents of this file are subject to the Mozilla Public License
//  Version 1.1 (the "License"); you may not use this file except in compliance
//  with the License. You may obtain a copy of the License at
//  
//  http://www.mozilla.org/MPL/
//  
//  Software distributed under the License is distributed on an "AS IS"
//  basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
//  License for the specific language governing rights and limitations under 
//  the License.
//  
//  The Initial Developer of the Original Code is Robert Smyth.
//  Portions created by Robert Smyth are Copyright (C) 2008.
//  
//  All Rights Reserved.

#endregion

using NUnitGrid.Framework.NUnitGrid;
using NUnitGrid.Framework.Pipes;


namespace NUnitGrid.Protocol.Commands
{
    public class GridMetricsUpdateTransaction : IProtocolTransaction
    {
        private readonly int busyNodesCount;
        private readonly ITaskProgressListener listener;
        private readonly int onlineNodesCount;
        private readonly int pendingTasksCount;
        private readonly IPipe pipe;
        private ProtocolAppCommandMessage commandMessage;

        public GridMetricsUpdateTransaction(IPipe pipe, ITaskProgressListener listener, int busyNodesCount,
                                            int onlineNodesCount, int pendingTasksCount)
        {
            this.pipe = pipe;
            this.listener = listener;
            this.busyNodesCount = busyNodesCount;
            this.onlineNodesCount = onlineNodesCount;
            this.pendingTasksCount = pendingTasksCount;
        }

        public void SendCommand()
        {
            commandMessage = new ProtocolAppCommandMessage("Write.GridMetricsUpdate");

            commandMessage.AddProperty("BusyNodesCount", busyNodesCount);
            commandMessage.AddProperty("OnlineNodesCount", onlineNodesCount);
            commandMessage.AddProperty("PendingTasksCount", pendingTasksCount);

            listener.OnTaskStarted(this);

            pipe.SendMessage(commandMessage.ToString());
        }

        public bool IsValidReponse(string responseMessage)
        {
            return commandMessage.IsResponseMessage(responseMessage);
        }

        public void ParseResponse(string response)
        {
            var parser = new ProtocolMessageParser(response);
            var nodeMachineName = parser.GetValue("MachineName");
            listener.OnTaskCompleted(this, true, nodeMachineName);
        }
    }
}